// compile_server模块

#pragma once

#include <iostream>

#include <unistd.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "../comm/util.hpp"
#include "../comm/log.hpp"

// 该服务负责代码的编译

namespace dsh_compiler
{
    // 引入路径拼接功能
    using namespace dsh_util;
    using namespace dsh_log;

    class Compiler
    {
    public:
        Compiler()
        {}
        ~Compiler()
        {}
        static bool Compile(const std::string &file_name)
        {
            // 编译成功返回true， 失败返回false
            pid_t pid = fork();
            if (pid < 0)
            {
                LOG(ERROR) << "创建子进程失败" << std::endl;
                return false; // 创建子进程失败
            }
            else if (pid == 0)
            {
                umask(0);
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(), O_CREAT | O_WRONLY, 0644);
                if (_stderr < 0)
                {
                    LOG(WARNING) << "形成compiler_error文件失败" << std::endl;
                    exit(2);
                }

                // 重定向到_stderr
                dup2(_stderr, 2);

                // 子进程：调用编译器
                // g++ -o exe src -std=c++11
                execlp("g++", "g++", "-o",
                        PathUtil::Exe(file_name).c_str(),
                        PathUtil::Src(file_name).c_str(),
                        "-std=c++11",
                        nullptr);

                LOG(ERROR) << "g++启动失败" << std::endl;
                // 进程替换失败就退出
                exit(1);
            }
            else
            {
                // 父进程
                waitpid(pid, nullptr, 0);
                // 判断是否成功，可以看文件是否生成
                if (FileUtil::isFileExits(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << "编译成功" << std::endl;
                    return true;
                }
            }
            LOG(ERROR) << "编译失败，没有形成可执行程序" << std::endl;
            return false;
        }
    };
}